package com.jxl.controller;

import com.jxl.service.WebSocketService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

import java.util.HashMap;
import java.util.Map;

/**
 * WebSocket测试Controller：通过HTTP接口触发消息发送，方便测试
 */
@RestController
@RequestMapping("/websocket/test") // 接口统一前缀
public class WebSocketTestController {

    // 注入业务层：复用之前写好的消息发送逻辑
    @Autowired
    private WebSocketService webSocketService;

    /**
     * 1. 发送广播消息：发给所有在线用户
     * 测试URL示例：http://localhost:8080/websocket/test/sendAll?message=所有在线用户请注意，这是一条广播消息
     */
    @GetMapping("/sendAll")
    public Map<String, Object> sendAllMessage(
            // 消息内容（必传）
            @RequestParam(required = true) String message
    ) {
        Map<String, Object> result = new HashMap<>();
        try {
            webSocketService.sendAll(message);
            result.put("code", 200);
            result.put("msg", "广播消息发送成功");
            result.put("data", "发送内容：" + message);
        } catch (Exception e) {
            result.put("code", 500);
            result.put("msg", "广播消息发送失败");
            result.put("error", e.getMessage());
        }
        return result;
    }

    /**
     * 2. 发送单点消息：发给指定单个用户
     * 测试URL示例：http://localhost:8080/websocket/test/sendOne?userId=1001&message=1001用户你好，这是你的专属消息
     */
    @GetMapping("/sendOne")
    public Map<String, Object> sendOneMessage(
            // 目标用户ID（必传，对应WebSocket连接时的userId）
            @RequestParam(required = true) String userId,
            // 消息内容（必传）
            @RequestParam(required = true) String message
    ) {
        Map<String, Object> result = new HashMap<>();
        try {
            webSocketService.sendOne(userId, message);
            result.put("code", 200);
            result.put("msg", "单点消息发送成功");
            result.put("data", "目标用户ID：" + userId + "，发送内容：" + message);
        } catch (Exception e) {
            result.put("code", 500);
            result.put("msg", "单点消息发送失败");
            result.put("error", e.getMessage());
        }
        return result;
    }

    /**
     * 3. 发送批量消息：发给多个指定用户
     * 测试URL示例：http://localhost:8080/websocket/test/sendMore?userIds=1001,1002,1003&message=1001/1002/1003用户，这是你们的批量消息
     */
    @GetMapping("/sendMore")
    public Map<String, Object> sendMoreMessage(
            // 目标用户ID列表（必传，多个ID用英文逗号分隔，如1001,1002）
            @RequestParam(required = true) String userIds,
            // 消息内容（必传）
            @RequestParam(required = true) String message
    ) {
        Map<String, Object> result = new HashMap<>();
        try {
            // 将逗号分隔的用户ID转成数组
            String[] userIdArray = userIds.split(",");
            webSocketService.sendMore(userIdArray, message);
            result.put("code", 200);
            result.put("msg", "批量消息发送成功");
            result.put("data", "目标用户IDs：" + userIds + "，发送内容：" + message);
        } catch (Exception e) {
            result.put("code", 500);
            result.put("msg", "批量消息发送失败");
            result.put("error", e.getMessage());
        }
        return result;
    }
}